import requests


class LoginApi():
    def __init__(self):
        self.login_url = 'http://ihrm-test.itheima.net/api/sys/login'

    def get_au_headers(self, body):
        resp = requests.post(self.login_url, json=body)
        headers = {
            'Authorization': resp.json().get('data')
        }
        return headers
